原始题目:剑指 Offer 54. 二叉搜索树的第k大节点 (opens new window)

解题思路:

基于二叉搜索树的特点,可以对二叉搜索树进行中序遍历,然后将第 kk 个节点的值返回即可。

代码:

public int kthLargest(TreeNode root, int k) {
    Deque<TreeNode> queue = new LinkedList<>();
    if (root != null) {
        queue.push(root);
    }
    int ans = 0;
    while (root != null || !queue.isEmpty()) {
        while (root != null) {
            queue.push(root);
            root = root.right;
        }
        root = queue.pop();
        if (--k == 0) {
            ans = root.val;
            break;
        }
        root = root.left;
    }
    return ans;
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20

复杂度分析

  • 时间复杂度O(N)O(N)NN 是二叉搜索树的节点数。最差情况下,k=Nk = N,需要遍历所有的节点。
  • 空间复杂度O(N)O(N):最差情况下,树退化成链表,需要存储所有的节点。
上次更新: 2023/10/15